package com.cuixichang.minimalism.core.mybatis.core;

import com.baomidou.mybatisplus.generator.config.builder.CustomFile;

import java.util.List;

public class DTOClassDecorate extends AbsClassDecorate{

    public DTOClassDecorate(ClassDecorate decorate) {
        super(decorate);
    }

    @Override
    public void definition(List<CustomFile> customFiles) {
        decorate.definition(customFiles);
        CustomFile.Builder builder = new CustomFile.Builder();
        CustomFile build = builder.templatePath("/template/DTO.java.vm")
                .enableFileOverride().fileName("DTO.java")
                .packageName("transfer.dto").build();
        customFiles.add(build);
    }
}
